import unittest
from selenium import webdriver
from time import sleep
from Page.Page import Page
from Page.Page2 import Page2
class MyTestCase(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Edge()
        self.driver.maximize_window()
        print('初始化')

    def tearDown(self):
        sleep(5)
        self.driver.quit()
        print('结束')

    def test_01(self):
        po = Page(self.driver)
        po.yingpian_luoji_01()

    def test_02(self):
        po = Page2(self.driver)
        po.yingpian_luoji_02()
if __name__ == '__main__':
    unittest.main()
